Generated by Cython 3.0.2
Yellow lines hint at Python interaction.
Click on a line that starts with a "+" to see the C code that Cython generated for it.
Raw output: blocks.c
+01: # cython: infer_types=True, wraparound=False, nonecheck=False, boundscheck=False, cdivision=True, language_level=3, profile=False, autogen_pxd=True
__pyx_t_7 = __Pyx_PyDict_NewPresized(0); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 1, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_7); if (PyDict_SetItem(__pyx_d, __pyx_n_s_test, __pyx_t_7) < 0) __PYX_ERR(0, 1, __pyx_L1_error) __Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
02:
+03: import numpy as np
__pyx_t_7 = __Pyx_ImportDottedModule(__pyx_n_s_numpy, NULL); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 3, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_7); if (PyDict_SetItem(__pyx_d, __pyx_n_s_np, __pyx_t_7) < 0) __PYX_ERR(0, 3, __pyx_L1_error) __Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
04: cimport numpy as np
05:
+06: def assemble_frame_from_blocks(blocks_stack: np.ndarray, n_rows: int, n_cols: int):
/* Python wrapper */
static PyObject *__pyx_pw_7nanopyx_4core_9transform_6blocks_1assemble_frame_from_blocks(PyObject *__pyx_self,
#if CYTHON_METH_FASTCALL
PyObject *const *__pyx_args, Py_ssize_t __pyx_nargs, PyObject *__pyx_kwds
#else
PyObject *__pyx_args, PyObject *__pyx_kwds
#endif
); /*proto*/
static PyMethodDef __pyx_mdef_7nanopyx_4core_9transform_6blocks_1assemble_frame_from_blocks = {"assemble_frame_from_blocks", (PyCFunction)(void*)(__Pyx_PyCFunction_FastCallWithKeywords)__pyx_pw_7nanopyx_4core_9transform_6blocks_1assemble_frame_from_blocks, __Pyx_METH_FASTCALL|METH_KEYWORDS, 0};
static PyObject *__pyx_pw_7nanopyx_4core_9transform_6blocks_1assemble_frame_from_blocks(PyObject *__pyx_self,
#if CYTHON_METH_FASTCALL
PyObject *const *__pyx_args, Py_ssize_t __pyx_nargs, PyObject *__pyx_kwds
#else
PyObject *__pyx_args, PyObject *__pyx_kwds
#endif
) {
PyArrayObject *__pyx_v_blocks_stack = 0;
PyObject *__pyx_v_n_rows = 0;
PyObject *__pyx_v_n_cols = 0;
#if !CYTHON_METH_FASTCALL
CYTHON_UNUSED Py_ssize_t __pyx_nargs;
#endif
CYTHON_UNUSED PyObject *const *__pyx_kwvalues;
PyObject *__pyx_r = 0;
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("assemble_frame_from_blocks (wrapper)", 0);
#if !CYTHON_METH_FASTCALL
#if CYTHON_ASSUME_SAFE_MACROS
__pyx_nargs = PyTuple_GET_SIZE(__pyx_args);
#else
__pyx_nargs = PyTuple_Size(__pyx_args);
if (unlikely((__pyx_nargs < 0))) __PYX_ERR(0, 6, __pyx_L3_error)
#endif
#endif
__pyx_kwvalues = __Pyx_KwValues_FASTCALL(__pyx_args, __pyx_nargs);
{
PyObject **__pyx_pyargnames[] = {&__pyx_n_s_blocks_stack,&__pyx_n_s_n_rows,&__pyx_n_s_n_cols,0};
PyObject* values[3] = {0,0,0};
if (__pyx_kwds) {
Py_ssize_t kw_args;
switch (__pyx_nargs) {
case 3: values[2] = __Pyx_Arg_FASTCALL(__pyx_args, 2);
CYTHON_FALLTHROUGH;
case 2: values[1] = __Pyx_Arg_FASTCALL(__pyx_args, 1);
CYTHON_FALLTHROUGH;
case 1: values[0] = __Pyx_Arg_FASTCALL(__pyx_args, 0);
CYTHON_FALLTHROUGH;
case 0: break;
default: goto __pyx_L5_argtuple_error;
}
kw_args = __Pyx_NumKwargs_FASTCALL(__pyx_kwds);
switch (__pyx_nargs) {
case 0:
if (likely((values[0] = __Pyx_GetKwValue_FASTCALL(__pyx_kwds, __pyx_kwvalues, __pyx_n_s_blocks_stack)) != 0)) {
(void)__Pyx_Arg_NewRef_FASTCALL(values[0]);
kw_args--;
}
else if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 6, __pyx_L3_error)
else goto __pyx_L5_argtuple_error;
CYTHON_FALLTHROUGH;
case 1:
if (likely((values[1] = __Pyx_GetKwValue_FASTCALL(__pyx_kwds, __pyx_kwvalues, __pyx_n_s_n_rows)) != 0)) {
(void)__Pyx_Arg_NewRef_FASTCALL(values[1]);
kw_args--;
}
else if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 6, __pyx_L3_error)
else {
__Pyx_RaiseArgtupleInvalid("assemble_frame_from_blocks", 1, 3, 3, 1); __PYX_ERR(0, 6, __pyx_L3_error)
}
CYTHON_FALLTHROUGH;
case 2:
if (likely((values[2] = __Pyx_GetKwValue_FASTCALL(__pyx_kwds, __pyx_kwvalues, __pyx_n_s_n_cols)) != 0)) {
(void)__Pyx_Arg_NewRef_FASTCALL(values[2]);
kw_args--;
}
else if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 6, __pyx_L3_error)
else {
__Pyx_RaiseArgtupleInvalid("assemble_frame_from_blocks", 1, 3, 3, 2); __PYX_ERR(0, 6, __pyx_L3_error)
}
}
if (unlikely(kw_args > 0)) {
const Py_ssize_t kwd_pos_args = __pyx_nargs;
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_kwvalues, __pyx_pyargnames, 0, values + 0, kwd_pos_args, "assemble_frame_from_blocks") < 0)) __PYX_ERR(0, 6, __pyx_L3_error)
}
} else if (unlikely(__pyx_nargs != 3)) {
goto __pyx_L5_argtuple_error;
} else {
values[0] = __Pyx_Arg_FASTCALL(__pyx_args, 0);
values[1] = __Pyx_Arg_FASTCALL(__pyx_args, 1);
values[2] = __Pyx_Arg_FASTCALL(__pyx_args, 2);
}
__pyx_v_blocks_stack = ((PyArrayObject *)values[0]);
__pyx_v_n_rows = ((PyObject*)values[1]);
__pyx_v_n_cols = ((PyObject*)values[2]);
}
goto __pyx_L4_argument_unpacking_done;
__pyx_L5_argtuple_error:;
__Pyx_RaiseArgtupleInvalid("assemble_frame_from_blocks", 1, 3, 3, __pyx_nargs); __PYX_ERR(0, 6, __pyx_L3_error)
goto __pyx_L3_error;
__pyx_L3_error:;
{
Py_ssize_t __pyx_temp;
for (__pyx_temp=0; __pyx_temp < (Py_ssize_t)(sizeof(values)/sizeof(values[0])); ++__pyx_temp) {
__Pyx_Arg_XDECREF_FASTCALL(values[__pyx_temp]);
}
}
__Pyx_AddTraceback("nanopyx.core.transform.blocks.assemble_frame_from_blocks", __pyx_clineno, __pyx_lineno, __pyx_filename);
__Pyx_RefNannyFinishContext();
return NULL;
__pyx_L4_argument_unpacking_done:;
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_blocks_stack), __pyx_ptype_5numpy_ndarray, 0, "blocks_stack", 0))) __PYX_ERR(0, 6, __pyx_L1_error)
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_n_rows), (&PyInt_Type), 0, "n_rows", 1))) __PYX_ERR(0, 6, __pyx_L1_error)
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_n_cols), (&PyInt_Type), 0, "n_cols", 1))) __PYX_ERR(0, 6, __pyx_L1_error)
__pyx_r = __pyx_pf_7nanopyx_4core_9transform_6blocks_assemble_frame_from_blocks(__pyx_self, __pyx_v_blocks_stack, __pyx_v_n_rows, __pyx_v_n_cols);
int __pyx_lineno = 0;
const char *__pyx_filename = NULL;
int __pyx_clineno = 0;
/* function exit code */
goto __pyx_L0;
__pyx_L1_error:;
__pyx_r = NULL;
__pyx_L0:;
{
Py_ssize_t __pyx_temp;
for (__pyx_temp=0; __pyx_temp < (Py_ssize_t)(sizeof(values)/sizeof(values[0])); ++__pyx_temp) {
__Pyx_Arg_XDECREF_FASTCALL(values[__pyx_temp]);
}
}
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
static PyObject *__pyx_pf_7nanopyx_4core_9transform_6blocks_assemble_frame_from_blocks(CYTHON_UNUSED PyObject *__pyx_self, PyArrayObject *__pyx_v_blocks_stack, PyObject *__pyx_v_n_rows, PyObject *__pyx_v_n_cols) {
PyObject *__pyx_r = NULL;
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("assemble_frame_from_blocks", 0);
/* … */
/* function exit code */
__pyx_L1_error:;
__Pyx_XDECREF(__pyx_t_1);
__Pyx_XDECREF(__pyx_t_2);
__Pyx_XDECREF(__pyx_t_3);
__Pyx_XDECREF(__pyx_t_4);
__PYX_XCLEAR_MEMVIEW(&__pyx_t_6, 1);
__PYX_XCLEAR_MEMVIEW(&__pyx_t_8, 1);
__Pyx_AddTraceback("nanopyx.core.transform.blocks.assemble_frame_from_blocks", __pyx_clineno, __pyx_lineno, __pyx_filename);
__pyx_r = NULL;
__pyx_L0:;
__Pyx_XGIVEREF(__pyx_r);
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* … */
__pyx_tuple__22 = PyTuple_Pack(3, __pyx_n_s_blocks_stack, __pyx_n_s_n_rows, __pyx_n_s_n_cols); if (unlikely(!__pyx_tuple__22)) __PYX_ERR(0, 6, __pyx_L1_error)
__Pyx_GOTREF(__pyx_tuple__22);
__Pyx_GIVEREF(__pyx_tuple__22);
/* … */
__pyx_t_7 = __Pyx_PyDict_NewPresized(3); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 6, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
if (PyDict_SetItem(__pyx_t_7, __pyx_n_s_blocks_stack, __pyx_kp_s_np_ndarray) < 0) __PYX_ERR(0, 6, __pyx_L1_error)
if (PyDict_SetItem(__pyx_t_7, __pyx_n_s_n_rows, __pyx_n_s_int) < 0) __PYX_ERR(0, 6, __pyx_L1_error)
if (PyDict_SetItem(__pyx_t_7, __pyx_n_s_n_cols, __pyx_n_s_int) < 0) __PYX_ERR(0, 6, __pyx_L1_error)
__pyx_t_4 = __Pyx_CyFunction_New(&__pyx_mdef_7nanopyx_4core_9transform_6blocks_1assemble_frame_from_blocks, 0, __pyx_n_s_assemble_frame_from_blocks, NULL, __pyx_n_s_nanopyx_core_transform_blocks, __pyx_d, ((PyObject *)__pyx_codeobj__23)); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 6, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_4);
__Pyx_CyFunction_SetAnnotationsDict(__pyx_t_4, __pyx_t_7);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
if (PyDict_SetItem(__pyx_d, __pyx_n_s_assemble_frame_from_blocks, __pyx_t_4) < 0) __PYX_ERR(0, 6, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
__pyx_codeobj__23 = (PyObject*)__Pyx_PyCode_New(3, 0, 0, 3, 0, CO_OPTIMIZED|CO_NEWLOCALS, __pyx_empty_bytes, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_tuple__22, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_kp_s_src_nanopyx_core_transform_block, __pyx_n_s_assemble_frame_from_blocks, 6, __pyx_empty_bytes); if (unlikely(!__pyx_codeobj__23)) __PYX_ERR(0, 6, __pyx_L1_error)
+07: return _assemble_frame_from_blocks(blocks_stack.astype(np.float32), n_rows, n_cols)
__Pyx_XDECREF(__pyx_r); __pyx_t_2 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_blocks_stack), __pyx_n_s_astype); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 7, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_2); __Pyx_GetModuleGlobalName(__pyx_t_3, __pyx_n_s_np); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 7, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_3); __pyx_t_4 = __Pyx_PyObject_GetAttrStr(__pyx_t_3, __pyx_n_s_float32); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 7, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_4); __Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0; __pyx_t_3 = NULL; __pyx_t_5 = 0; #if CYTHON_UNPACK_METHODS if (likely(PyMethod_Check(__pyx_t_2))) { __pyx_t_3 = PyMethod_GET_SELF(__pyx_t_2); if (likely(__pyx_t_3)) { PyObject* function = PyMethod_GET_FUNCTION(__pyx_t_2); __Pyx_INCREF(__pyx_t_3); __Pyx_INCREF(function); __Pyx_DECREF_SET(__pyx_t_2, function); __pyx_t_5 = 1; } } #endif { PyObject *__pyx_callargs[2] = {__pyx_t_3, __pyx_t_4}; __pyx_t_1 = __Pyx_PyObject_FastCall(__pyx_t_2, __pyx_callargs+1-__pyx_t_5, 1+__pyx_t_5); __Pyx_XDECREF(__pyx_t_3); __pyx_t_3 = 0; __Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0; if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 7, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); __Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0; } __pyx_t_6 = __Pyx_PyObject_to_MemoryviewSlice_dsdsds_float(__pyx_t_1, PyBUF_WRITABLE); if (unlikely(!__pyx_t_6.memview)) __PYX_ERR(0, 7, __pyx_L1_error) __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0; __pyx_t_5 = __Pyx_PyInt_As_int(__pyx_v_n_rows); if (unlikely((__pyx_t_5 == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 7, __pyx_L1_error) __pyx_t_7 = __Pyx_PyInt_As_int(__pyx_v_n_cols); if (unlikely((__pyx_t_7 == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 7, __pyx_L1_error) __pyx_t_8 = __pyx_f_7nanopyx_4core_9transform_6blocks__assemble_frame_from_blocks(__pyx_t_6, __pyx_t_5, __pyx_t_7); if (unlikely(!__pyx_t_8.memview)) __PYX_ERR(0, 7, __pyx_L1_error) __PYX_XCLEAR_MEMVIEW(&__pyx_t_6, 1); __pyx_t_6.memview = NULL; __pyx_t_6.data = NULL; __pyx_t_1 = __pyx_memoryview_fromslice(__pyx_t_8, 2, (PyObject *(*)(char *)) __pyx_memview_get_float, (int (*)(char *, PyObject *)) __pyx_memview_set_float, 0);; if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 7, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); __PYX_XCLEAR_MEMVIEW(&__pyx_t_8, 1); __pyx_t_8.memview = NULL; __pyx_t_8.data = NULL; __pyx_r = __pyx_t_1; __pyx_t_1 = 0; goto __pyx_L0;
08:
+09: cdef float[:, :] _assemble_frame_from_blocks(float[:, :, :] blocks_stack, int n_rows, int n_cols) nogil:
static __Pyx_memviewslice __pyx_f_7nanopyx_4core_9transform_6blocks__assemble_frame_from_blocks(__Pyx_memviewslice __pyx_v_blocks_stack, int __pyx_v_n_rows, int __pyx_v_n_cols) {
int __pyx_v_block_cols_len;
int __pyx_v_block_rows_len;
int __pyx_v_cols_len;
int __pyx_v_rows_len;
int __pyx_v_row_i;
int __pyx_v_col_i;
int __pyx_v_count;
__Pyx_memviewslice __pyx_v_reconstructed_image = { 0, 0, { 0 }, { 0 }, { 0 } };
__Pyx_memviewslice __pyx_r = { 0, 0, { 0 }, { 0 }, { 0 } };
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("_assemble_frame_from_blocks", 1);
/* … */
/* function exit code */
__pyx_L1_error:;
#ifdef WITH_THREAD
__pyx_gilstate_save = __Pyx_PyGILState_Ensure();
#endif
__Pyx_XDECREF(__pyx_t_2);
__Pyx_XDECREF(__pyx_t_3);
__Pyx_XDECREF(__pyx_t_4);
__Pyx_XDECREF(__pyx_t_5);
__Pyx_XDECREF(__pyx_t_6);
__PYX_XCLEAR_MEMVIEW(&__pyx_t_7, 1);
__PYX_XCLEAR_MEMVIEW(&__pyx_t_14, 1);
__pyx_r.data = NULL;
__pyx_r.memview = NULL;
__Pyx_AddTraceback("nanopyx.core.transform.blocks._assemble_frame_from_blocks", __pyx_clineno, __pyx_lineno, __pyx_filename);
goto __pyx_L2;
__pyx_L0:;
if (unlikely(!__pyx_r.memview)) {
#ifdef WITH_THREAD
PyGILState_STATE __pyx_gilstate_save = __Pyx_PyGILState_Ensure();
#endif
PyErr_SetString(PyExc_TypeError, "Memoryview return value is not initialized");
#ifdef WITH_THREAD
__Pyx_PyGILState_Release(__pyx_gilstate_save);
#endif
}
#ifdef WITH_THREAD
__Pyx_PyGILState_Release(__pyx_gilstate_save);
#endif
__pyx_L2:;
#ifdef WITH_THREAD
__pyx_gilstate_save = __Pyx_PyGILState_Ensure();
#endif
__PYX_XCLEAR_MEMVIEW(&__pyx_v_reconstructed_image, 1);
__Pyx_RefNannyFinishContext();
#ifdef WITH_THREAD
__Pyx_PyGILState_Release(__pyx_gilstate_save);
#endif
return __pyx_r;
}
10:
+11: cdef int block_cols_len = blocks_stack.shape[2]
__pyx_v_block_cols_len = (__pyx_v_blocks_stack.shape[2]);
+12: cdef int block_rows_len = blocks_stack.shape[1]
__pyx_v_block_rows_len = (__pyx_v_blocks_stack.shape[1]);
+13: cdef int cols_len = block_cols_len * n_cols
__pyx_v_cols_len = (__pyx_v_block_cols_len * __pyx_v_n_cols);
+14: cdef int rows_len = block_rows_len * n_rows
__pyx_v_rows_len = (__pyx_v_block_rows_len * __pyx_v_n_rows);
15: cdef int row_i, col_i, count
16: cdef float[:, :] reconstructed_image
17:
+18: with gil:
{
#ifdef WITH_THREAD
PyGILState_STATE __pyx_gilstate_save = __Pyx_PyGILState_Ensure();
#endif
/*try:*/ {
/* … */
/*finally:*/ {
/*normal exit:*/{
#ifdef WITH_THREAD
__Pyx_PyGILState_Release(__pyx_gilstate_save);
#endif
goto __pyx_L5;
}
__pyx_L4_error: {
#ifdef WITH_THREAD
__Pyx_PyGILState_Release(__pyx_gilstate_save);
#endif
goto __pyx_L1_error;
}
__pyx_L5:;
}
}
+19: assert blocks_stack.shape[0] == n_rows * n_cols
#ifndef CYTHON_WITHOUT_ASSERTIONS
if (unlikely(__pyx_assertions_enabled())) {
__pyx_t_1 = ((__pyx_v_blocks_stack.shape[0]) == (__pyx_v_n_rows * __pyx_v_n_cols));
if (unlikely(!__pyx_t_1)) {
__Pyx_Raise(__pyx_builtin_AssertionError, 0, 0, 0);
__PYX_ERR(0, 19, __pyx_L4_error)
}
}
#else
if ((1)); else __PYX_ERR(0, 19, __pyx_L4_error)
#endif
+20: reconstructed_image = np.empty((rows_len, cols_len), dtype=np.float32)
__Pyx_GetModuleGlobalName(__pyx_t_2, __pyx_n_s_np); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 20, __pyx_L4_error) __Pyx_GOTREF(__pyx_t_2); __pyx_t_3 = __Pyx_PyObject_GetAttrStr(__pyx_t_2, __pyx_n_s_empty); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 20, __pyx_L4_error) __Pyx_GOTREF(__pyx_t_3); __Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0; __pyx_t_2 = __Pyx_PyInt_From_int(__pyx_v_rows_len); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 20, __pyx_L4_error) __Pyx_GOTREF(__pyx_t_2); __pyx_t_4 = __Pyx_PyInt_From_int(__pyx_v_cols_len); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 20, __pyx_L4_error) __Pyx_GOTREF(__pyx_t_4); __pyx_t_5 = PyTuple_New(2); if (unlikely(!__pyx_t_5)) __PYX_ERR(0, 20, __pyx_L4_error) __Pyx_GOTREF(__pyx_t_5); __Pyx_GIVEREF(__pyx_t_2); if (__Pyx_PyTuple_SET_ITEM(__pyx_t_5, 0, __pyx_t_2)) __PYX_ERR(0, 20, __pyx_L4_error); __Pyx_GIVEREF(__pyx_t_4); if (__Pyx_PyTuple_SET_ITEM(__pyx_t_5, 1, __pyx_t_4)) __PYX_ERR(0, 20, __pyx_L4_error); __pyx_t_2 = 0; __pyx_t_4 = 0; __pyx_t_4 = PyTuple_New(1); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 20, __pyx_L4_error) __Pyx_GOTREF(__pyx_t_4); __Pyx_GIVEREF(__pyx_t_5); if (__Pyx_PyTuple_SET_ITEM(__pyx_t_4, 0, __pyx_t_5)) __PYX_ERR(0, 20, __pyx_L4_error); __pyx_t_5 = 0; __pyx_t_5 = __Pyx_PyDict_NewPresized(1); if (unlikely(!__pyx_t_5)) __PYX_ERR(0, 20, __pyx_L4_error) __Pyx_GOTREF(__pyx_t_5); __Pyx_GetModuleGlobalName(__pyx_t_2, __pyx_n_s_np); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 20, __pyx_L4_error) __Pyx_GOTREF(__pyx_t_2); __pyx_t_6 = __Pyx_PyObject_GetAttrStr(__pyx_t_2, __pyx_n_s_float32); if (unlikely(!__pyx_t_6)) __PYX_ERR(0, 20, __pyx_L4_error) __Pyx_GOTREF(__pyx_t_6); __Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0; if (PyDict_SetItem(__pyx_t_5, __pyx_n_s_dtype, __pyx_t_6) < 0) __PYX_ERR(0, 20, __pyx_L4_error) __Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0; __pyx_t_6 = __Pyx_PyObject_Call(__pyx_t_3, __pyx_t_4, __pyx_t_5); if (unlikely(!__pyx_t_6)) __PYX_ERR(0, 20, __pyx_L4_error) __Pyx_GOTREF(__pyx_t_6); __Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0; __Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0; __Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0; __pyx_t_7 = __Pyx_PyObject_to_MemoryviewSlice_dsds_float(__pyx_t_6, PyBUF_WRITABLE); if (unlikely(!__pyx_t_7.memview)) __PYX_ERR(0, 20, __pyx_L4_error) __Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0; __pyx_v_reconstructed_image = __pyx_t_7; __pyx_t_7.memview = NULL; __pyx_t_7.data = NULL;
+21: assert reconstructed_image.shape[0] == rows_len
#ifndef CYTHON_WITHOUT_ASSERTIONS
if (unlikely(__pyx_assertions_enabled())) {
__pyx_t_1 = ((__pyx_v_reconstructed_image.shape[0]) == __pyx_v_rows_len);
if (unlikely(!__pyx_t_1)) {
__Pyx_Raise(__pyx_builtin_AssertionError, 0, 0, 0);
__PYX_ERR(0, 21, __pyx_L4_error)
}
}
#else
if ((1)); else __PYX_ERR(0, 21, __pyx_L4_error)
#endif
+22: assert reconstructed_image.shape[1] == cols_len
#ifndef CYTHON_WITHOUT_ASSERTIONS
if (unlikely(__pyx_assertions_enabled())) {
__pyx_t_1 = ((__pyx_v_reconstructed_image.shape[1]) == __pyx_v_cols_len);
if (unlikely(!__pyx_t_1)) {
__Pyx_Raise(__pyx_builtin_AssertionError, 0, 0, 0);
__PYX_ERR(0, 22, __pyx_L4_error)
}
}
#else
if ((1)); else __PYX_ERR(0, 22, __pyx_L4_error)
#endif
+23: count = 0
__pyx_v_count = 0;
+24: for row_i in range(n_rows):
__pyx_t_8 = __pyx_v_n_rows;
__pyx_t_9 = __pyx_t_8;
for (__pyx_t_10 = 0; __pyx_t_10 < __pyx_t_9; __pyx_t_10+=1) {
__pyx_v_row_i = __pyx_t_10;
+25: for col_i in range(n_cols):
__pyx_t_11 = __pyx_v_n_cols;
__pyx_t_12 = __pyx_t_11;
for (__pyx_t_13 = 0; __pyx_t_13 < __pyx_t_12; __pyx_t_13+=1) {
__pyx_v_col_i = __pyx_t_13;
+26: reconstructed_image[row_i*block_rows_len:(row_i+1)*block_rows_len, col_i*block_cols_len:(col_i+1)*block_cols_len] = blocks_stack[count]
__pyx_t_7.data = __pyx_v_blocks_stack.data;
__pyx_t_7.memview = __pyx_v_blocks_stack.memview;
__PYX_INC_MEMVIEW(&__pyx_t_7, 1);
{
Py_ssize_t __pyx_tmp_idx = __pyx_v_count;
Py_ssize_t __pyx_tmp_stride = __pyx_v_blocks_stack.strides[0];
__pyx_t_7.data += __pyx_tmp_idx * __pyx_tmp_stride;
}
__pyx_t_7.shape[0] = __pyx_v_blocks_stack.shape[1];
__pyx_t_7.strides[0] = __pyx_v_blocks_stack.strides[1];
__pyx_t_7.suboffsets[0] = -1;
__pyx_t_7.shape[1] = __pyx_v_blocks_stack.shape[2];
__pyx_t_7.strides[1] = __pyx_v_blocks_stack.strides[2];
__pyx_t_7.suboffsets[1] = -1;
__pyx_t_14.data = __pyx_v_reconstructed_image.data;
__pyx_t_14.memview = __pyx_v_reconstructed_image.memview;
__PYX_INC_MEMVIEW(&__pyx_t_14, 1);
__pyx_t_15 = -1;
if (unlikely(__pyx_memoryview_slice_memviewslice(
&__pyx_t_14,
__pyx_v_reconstructed_image.shape[0], __pyx_v_reconstructed_image.strides[0], __pyx_v_reconstructed_image.suboffsets[0],
0,
0,
&__pyx_t_15,
(__pyx_v_row_i * __pyx_v_block_rows_len),
((__pyx_v_row_i + 1) * __pyx_v_block_rows_len),
0,
1,
1,
0,
1) < 0))
{
__PYX_ERR(0, 26, __pyx_L4_error)
}
if (unlikely(__pyx_memoryview_slice_memviewslice(
&__pyx_t_14,
__pyx_v_reconstructed_image.shape[1], __pyx_v_reconstructed_image.strides[1], __pyx_v_reconstructed_image.suboffsets[1],
1,
1,
&__pyx_t_15,
(__pyx_v_col_i * __pyx_v_block_cols_len),
((__pyx_v_col_i + 1) * __pyx_v_block_cols_len),
0,
1,
1,
0,
1) < 0))
{
__PYX_ERR(0, 26, __pyx_L4_error)
}
if (unlikely((__pyx_memoryview_copy_contents(__pyx_t_7, __pyx_t_14, 2, 2, 0) < 0))) __PYX_ERR(0, 26, __pyx_L4_error)
__PYX_XCLEAR_MEMVIEW(&__pyx_t_14, 1);
__pyx_t_14.memview = NULL; __pyx_t_14.data = NULL;
__PYX_XCLEAR_MEMVIEW(&__pyx_t_7, 1);
__pyx_t_7.memview = NULL; __pyx_t_7.data = NULL;
+27: count += 1
__pyx_v_count = (__pyx_v_count + 1);
}
}
}
28:
+29: return reconstructed_image
__PYX_INC_MEMVIEW(&__pyx_v_reconstructed_image, 0); __pyx_r = __pyx_v_reconstructed_image; goto __pyx_L0;
30:
+31: def split_frame_into_blocks(img: np.ndarray, n_rows: int, n_cols: int):
/* Python wrapper */
static PyObject *__pyx_pw_7nanopyx_4core_9transform_6blocks_3split_frame_into_blocks(PyObject *__pyx_self,
#if CYTHON_METH_FASTCALL
PyObject *const *__pyx_args, Py_ssize_t __pyx_nargs, PyObject *__pyx_kwds
#else
PyObject *__pyx_args, PyObject *__pyx_kwds
#endif
); /*proto*/
static PyMethodDef __pyx_mdef_7nanopyx_4core_9transform_6blocks_3split_frame_into_blocks = {"split_frame_into_blocks", (PyCFunction)(void*)(__Pyx_PyCFunction_FastCallWithKeywords)__pyx_pw_7nanopyx_4core_9transform_6blocks_3split_frame_into_blocks, __Pyx_METH_FASTCALL|METH_KEYWORDS, 0};
static PyObject *__pyx_pw_7nanopyx_4core_9transform_6blocks_3split_frame_into_blocks(PyObject *__pyx_self,
#if CYTHON_METH_FASTCALL
PyObject *const *__pyx_args, Py_ssize_t __pyx_nargs, PyObject *__pyx_kwds
#else
PyObject *__pyx_args, PyObject *__pyx_kwds
#endif
) {
PyArrayObject *__pyx_v_img = 0;
PyObject *__pyx_v_n_rows = 0;
PyObject *__pyx_v_n_cols = 0;
#if !CYTHON_METH_FASTCALL
CYTHON_UNUSED Py_ssize_t __pyx_nargs;
#endif
CYTHON_UNUSED PyObject *const *__pyx_kwvalues;
PyObject *__pyx_r = 0;
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("split_frame_into_blocks (wrapper)", 0);
#if !CYTHON_METH_FASTCALL
#if CYTHON_ASSUME_SAFE_MACROS
__pyx_nargs = PyTuple_GET_SIZE(__pyx_args);
#else
__pyx_nargs = PyTuple_Size(__pyx_args);
if (unlikely((__pyx_nargs < 0))) __PYX_ERR(0, 31, __pyx_L3_error)
#endif
#endif
__pyx_kwvalues = __Pyx_KwValues_FASTCALL(__pyx_args, __pyx_nargs);
{
PyObject **__pyx_pyargnames[] = {&__pyx_n_s_img,&__pyx_n_s_n_rows,&__pyx_n_s_n_cols,0};
PyObject* values[3] = {0,0,0};
if (__pyx_kwds) {
Py_ssize_t kw_args;
switch (__pyx_nargs) {
case 3: values[2] = __Pyx_Arg_FASTCALL(__pyx_args, 2);
CYTHON_FALLTHROUGH;
case 2: values[1] = __Pyx_Arg_FASTCALL(__pyx_args, 1);
CYTHON_FALLTHROUGH;
case 1: values[0] = __Pyx_Arg_FASTCALL(__pyx_args, 0);
CYTHON_FALLTHROUGH;
case 0: break;
default: goto __pyx_L5_argtuple_error;
}
kw_args = __Pyx_NumKwargs_FASTCALL(__pyx_kwds);
switch (__pyx_nargs) {
case 0:
if (likely((values[0] = __Pyx_GetKwValue_FASTCALL(__pyx_kwds, __pyx_kwvalues, __pyx_n_s_img)) != 0)) {
(void)__Pyx_Arg_NewRef_FASTCALL(values[0]);
kw_args--;
}
else if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 31, __pyx_L3_error)
else goto __pyx_L5_argtuple_error;
CYTHON_FALLTHROUGH;
case 1:
if (likely((values[1] = __Pyx_GetKwValue_FASTCALL(__pyx_kwds, __pyx_kwvalues, __pyx_n_s_n_rows)) != 0)) {
(void)__Pyx_Arg_NewRef_FASTCALL(values[1]);
kw_args--;
}
else if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 31, __pyx_L3_error)
else {
__Pyx_RaiseArgtupleInvalid("split_frame_into_blocks", 1, 3, 3, 1); __PYX_ERR(0, 31, __pyx_L3_error)
}
CYTHON_FALLTHROUGH;
case 2:
if (likely((values[2] = __Pyx_GetKwValue_FASTCALL(__pyx_kwds, __pyx_kwvalues, __pyx_n_s_n_cols)) != 0)) {
(void)__Pyx_Arg_NewRef_FASTCALL(values[2]);
kw_args--;
}
else if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 31, __pyx_L3_error)
else {
__Pyx_RaiseArgtupleInvalid("split_frame_into_blocks", 1, 3, 3, 2); __PYX_ERR(0, 31, __pyx_L3_error)
}
}
if (unlikely(kw_args > 0)) {
const Py_ssize_t kwd_pos_args = __pyx_nargs;
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_kwvalues, __pyx_pyargnames, 0, values + 0, kwd_pos_args, "split_frame_into_blocks") < 0)) __PYX_ERR(0, 31, __pyx_L3_error)
}
} else if (unlikely(__pyx_nargs != 3)) {
goto __pyx_L5_argtuple_error;
} else {
values[0] = __Pyx_Arg_FASTCALL(__pyx_args, 0);
values[1] = __Pyx_Arg_FASTCALL(__pyx_args, 1);
values[2] = __Pyx_Arg_FASTCALL(__pyx_args, 2);
}
__pyx_v_img = ((PyArrayObject *)values[0]);
__pyx_v_n_rows = ((PyObject*)values[1]);
__pyx_v_n_cols = ((PyObject*)values[2]);
}
goto __pyx_L4_argument_unpacking_done;
__pyx_L5_argtuple_error:;
__Pyx_RaiseArgtupleInvalid("split_frame_into_blocks", 1, 3, 3, __pyx_nargs); __PYX_ERR(0, 31, __pyx_L3_error)
goto __pyx_L3_error;
__pyx_L3_error:;
{
Py_ssize_t __pyx_temp;
for (__pyx_temp=0; __pyx_temp < (Py_ssize_t)(sizeof(values)/sizeof(values[0])); ++__pyx_temp) {
__Pyx_Arg_XDECREF_FASTCALL(values[__pyx_temp]);
}
}
__Pyx_AddTraceback("nanopyx.core.transform.blocks.split_frame_into_blocks", __pyx_clineno, __pyx_lineno, __pyx_filename);
__Pyx_RefNannyFinishContext();
return NULL;
__pyx_L4_argument_unpacking_done:;
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_img), __pyx_ptype_5numpy_ndarray, 0, "img", 0))) __PYX_ERR(0, 31, __pyx_L1_error)
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_n_rows), (&PyInt_Type), 0, "n_rows", 1))) __PYX_ERR(0, 31, __pyx_L1_error)
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_n_cols), (&PyInt_Type), 0, "n_cols", 1))) __PYX_ERR(0, 31, __pyx_L1_error)
__pyx_r = __pyx_pf_7nanopyx_4core_9transform_6blocks_2split_frame_into_blocks(__pyx_self, __pyx_v_img, __pyx_v_n_rows, __pyx_v_n_cols);
int __pyx_lineno = 0;
const char *__pyx_filename = NULL;
int __pyx_clineno = 0;
/* function exit code */
goto __pyx_L0;
__pyx_L1_error:;
__pyx_r = NULL;
__pyx_L0:;
{
Py_ssize_t __pyx_temp;
for (__pyx_temp=0; __pyx_temp < (Py_ssize_t)(sizeof(values)/sizeof(values[0])); ++__pyx_temp) {
__Pyx_Arg_XDECREF_FASTCALL(values[__pyx_temp]);
}
}
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
static PyObject *__pyx_pf_7nanopyx_4core_9transform_6blocks_2split_frame_into_blocks(CYTHON_UNUSED PyObject *__pyx_self, PyArrayObject *__pyx_v_img, PyObject *__pyx_v_n_rows, PyObject *__pyx_v_n_cols) {
PyObject *__pyx_r = NULL;
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("split_frame_into_blocks", 0);
/* … */
/* function exit code */
__pyx_L1_error:;
__PYX_XCLEAR_MEMVIEW(&__pyx_t_1, 1);
__PYX_XCLEAR_MEMVIEW(&__pyx_t_4, 1);
__Pyx_XDECREF(__pyx_t_5);
__Pyx_AddTraceback("nanopyx.core.transform.blocks.split_frame_into_blocks", __pyx_clineno, __pyx_lineno, __pyx_filename);
__pyx_r = NULL;
__pyx_L0:;
__Pyx_XGIVEREF(__pyx_r);
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* … */
__pyx_tuple__24 = PyTuple_Pack(3, __pyx_n_s_img, __pyx_n_s_n_rows, __pyx_n_s_n_cols); if (unlikely(!__pyx_tuple__24)) __PYX_ERR(0, 31, __pyx_L1_error)
__Pyx_GOTREF(__pyx_tuple__24);
__Pyx_GIVEREF(__pyx_tuple__24);
/* … */
__pyx_t_4 = __Pyx_PyDict_NewPresized(3); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 31, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_4);
if (PyDict_SetItem(__pyx_t_4, __pyx_n_s_img, __pyx_kp_s_np_ndarray) < 0) __PYX_ERR(0, 31, __pyx_L1_error)
if (PyDict_SetItem(__pyx_t_4, __pyx_n_s_n_rows, __pyx_n_s_int) < 0) __PYX_ERR(0, 31, __pyx_L1_error)
if (PyDict_SetItem(__pyx_t_4, __pyx_n_s_n_cols, __pyx_n_s_int) < 0) __PYX_ERR(0, 31, __pyx_L1_error)
__pyx_t_7 = __Pyx_CyFunction_New(&__pyx_mdef_7nanopyx_4core_9transform_6blocks_3split_frame_into_blocks, 0, __pyx_n_s_split_frame_into_blocks, NULL, __pyx_n_s_nanopyx_core_transform_blocks, __pyx_d, ((PyObject *)__pyx_codeobj__25)); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 31, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__Pyx_CyFunction_SetAnnotationsDict(__pyx_t_7, __pyx_t_4);
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
if (PyDict_SetItem(__pyx_d, __pyx_n_s_split_frame_into_blocks, __pyx_t_7) < 0) __PYX_ERR(0, 31, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
+32: return _split_frame_into_blocks(img, n_rows, n_cols)
__Pyx_XDECREF(__pyx_r); __pyx_t_1 = __Pyx_PyObject_to_MemoryviewSlice_dsds_float(((PyObject *)__pyx_v_img), PyBUF_WRITABLE); if (unlikely(!__pyx_t_1.memview)) __PYX_ERR(0, 32, __pyx_L1_error) __pyx_t_2 = __Pyx_PyInt_As_int(__pyx_v_n_rows); if (unlikely((__pyx_t_2 == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 32, __pyx_L1_error) __pyx_t_3 = __Pyx_PyInt_As_int(__pyx_v_n_cols); if (unlikely((__pyx_t_3 == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 32, __pyx_L1_error) __pyx_t_4 = __pyx_f_7nanopyx_4core_9transform_6blocks__split_frame_into_blocks(__pyx_t_1, __pyx_t_2, __pyx_t_3); if (unlikely(!__pyx_t_4.memview)) __PYX_ERR(0, 32, __pyx_L1_error) __PYX_XCLEAR_MEMVIEW(&__pyx_t_1, 1); __pyx_t_1.memview = NULL; __pyx_t_1.data = NULL; __pyx_t_5 = __pyx_memoryview_fromslice(__pyx_t_4, 3, (PyObject *(*)(char *)) __pyx_memview_get_float, (int (*)(char *, PyObject *)) __pyx_memview_set_float, 0);; if (unlikely(!__pyx_t_5)) __PYX_ERR(0, 32, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_5); __PYX_XCLEAR_MEMVIEW(&__pyx_t_4, 1); __pyx_t_4.memview = NULL; __pyx_t_4.data = NULL; __pyx_r = __pyx_t_5; __pyx_t_5 = 0; goto __pyx_L0;
33:
+34: cdef float[:, :, :] _split_frame_into_blocks(float[:, :] img, int n_rows, int n_cols) nogil:
static __Pyx_memviewslice __pyx_f_7nanopyx_4core_9transform_6blocks__split_frame_into_blocks(__Pyx_memviewslice __pyx_v_img, int __pyx_v_n_rows, int __pyx_v_n_cols) {
int __pyx_v_block_cols_len;
int __pyx_v_block_rows_len;
int __pyx_v_cols_len;
int __pyx_v_rows_len;
int __pyx_v_row_i;
int __pyx_v_col_i;
int __pyx_v_count;
__Pyx_memviewslice __pyx_v_blocks_stack = { 0, 0, { 0 }, { 0 }, { 0 } };
__Pyx_memviewslice __pyx_r = { 0, 0, { 0 }, { 0 }, { 0 } };
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("_split_frame_into_blocks", 1);
/* … */
/* function exit code */
__pyx_L1_error:;
#ifdef WITH_THREAD
__pyx_gilstate_save = __Pyx_PyGILState_Ensure();
#endif
__Pyx_XDECREF(__pyx_t_2);
__Pyx_XDECREF(__pyx_t_3);
__Pyx_XDECREF(__pyx_t_4);
__Pyx_XDECREF(__pyx_t_5);
__Pyx_XDECREF(__pyx_t_6);
__PYX_XCLEAR_MEMVIEW(&__pyx_t_7, 1);
__PYX_XCLEAR_MEMVIEW(&__pyx_t_14, 1);
__PYX_XCLEAR_MEMVIEW(&__pyx_t_16, 1);
__pyx_r.data = NULL;
__pyx_r.memview = NULL;
__Pyx_AddTraceback("nanopyx.core.transform.blocks._split_frame_into_blocks", __pyx_clineno, __pyx_lineno, __pyx_filename);
goto __pyx_L2;
__pyx_L0:;
if (unlikely(!__pyx_r.memview)) {
#ifdef WITH_THREAD
PyGILState_STATE __pyx_gilstate_save = __Pyx_PyGILState_Ensure();
#endif
PyErr_SetString(PyExc_TypeError, "Memoryview return value is not initialized");
#ifdef WITH_THREAD
__Pyx_PyGILState_Release(__pyx_gilstate_save);
#endif
}
#ifdef WITH_THREAD
__Pyx_PyGILState_Release(__pyx_gilstate_save);
#endif
__pyx_L2:;
#ifdef WITH_THREAD
__pyx_gilstate_save = __Pyx_PyGILState_Ensure();
#endif
__PYX_XCLEAR_MEMVIEW(&__pyx_v_blocks_stack, 1);
__Pyx_RefNannyFinishContext();
#ifdef WITH_THREAD
__Pyx_PyGILState_Release(__pyx_gilstate_save);
#endif
return __pyx_r;
}
35:
+36: cdef int block_cols_len = img.shape[1] // n_cols
__pyx_v_block_cols_len = ((__pyx_v_img.shape[1]) / __pyx_v_n_cols);
+37: cdef int block_rows_len = img.shape[0] // n_rows
__pyx_v_block_rows_len = ((__pyx_v_img.shape[0]) / __pyx_v_n_rows);
+38: cdef int cols_len = block_cols_len * n_cols
__pyx_v_cols_len = (__pyx_v_block_cols_len * __pyx_v_n_cols);
+39: cdef int rows_len = block_rows_len * n_rows
__pyx_v_rows_len = (__pyx_v_block_rows_len * __pyx_v_n_rows);
40: cdef int row_i, col_i, count
41: cdef float[:, :, :] blocks_stack
42:
+43: with gil:
{
#ifdef WITH_THREAD
PyGILState_STATE __pyx_gilstate_save = __Pyx_PyGILState_Ensure();
#endif
/*try:*/ {
/* … */
/*finally:*/ {
/*normal exit:*/{
#ifdef WITH_THREAD
__Pyx_PyGILState_Release(__pyx_gilstate_save);
#endif
goto __pyx_L5;
}
__pyx_L4_error: {
#ifdef WITH_THREAD
__Pyx_PyGILState_Release(__pyx_gilstate_save);
#endif
goto __pyx_L1_error;
}
__pyx_L5:;
}
}
+44: assert img.shape[0] == rows_len
#ifndef CYTHON_WITHOUT_ASSERTIONS
if (unlikely(__pyx_assertions_enabled())) {
__pyx_t_1 = ((__pyx_v_img.shape[0]) == __pyx_v_rows_len);
if (unlikely(!__pyx_t_1)) {
__Pyx_Raise(__pyx_builtin_AssertionError, 0, 0, 0);
__PYX_ERR(0, 44, __pyx_L4_error)
}
}
#else
if ((1)); else __PYX_ERR(0, 44, __pyx_L4_error)
#endif
+45: assert img.shape[1] == cols_len
#ifndef CYTHON_WITHOUT_ASSERTIONS
if (unlikely(__pyx_assertions_enabled())) {
__pyx_t_1 = ((__pyx_v_img.shape[1]) == __pyx_v_cols_len);
if (unlikely(!__pyx_t_1)) {
__Pyx_Raise(__pyx_builtin_AssertionError, 0, 0, 0);
__PYX_ERR(0, 45, __pyx_L4_error)
}
}
#else
if ((1)); else __PYX_ERR(0, 45, __pyx_L4_error)
#endif
+46: blocks_stack = np.empty((n_rows * n_cols, block_rows_len, block_cols_len), dtype=np.float32)
__Pyx_GetModuleGlobalName(__pyx_t_2, __pyx_n_s_np); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 46, __pyx_L4_error) __Pyx_GOTREF(__pyx_t_2); __pyx_t_3 = __Pyx_PyObject_GetAttrStr(__pyx_t_2, __pyx_n_s_empty); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 46, __pyx_L4_error) __Pyx_GOTREF(__pyx_t_3); __Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0; __pyx_t_2 = __Pyx_PyInt_From_int((__pyx_v_n_rows * __pyx_v_n_cols)); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 46, __pyx_L4_error) __Pyx_GOTREF(__pyx_t_2); __pyx_t_4 = __Pyx_PyInt_From_int(__pyx_v_block_rows_len); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 46, __pyx_L4_error) __Pyx_GOTREF(__pyx_t_4); __pyx_t_5 = __Pyx_PyInt_From_int(__pyx_v_block_cols_len); if (unlikely(!__pyx_t_5)) __PYX_ERR(0, 46, __pyx_L4_error) __Pyx_GOTREF(__pyx_t_5); __pyx_t_6 = PyTuple_New(3); if (unlikely(!__pyx_t_6)) __PYX_ERR(0, 46, __pyx_L4_error) __Pyx_GOTREF(__pyx_t_6); __Pyx_GIVEREF(__pyx_t_2); if (__Pyx_PyTuple_SET_ITEM(__pyx_t_6, 0, __pyx_t_2)) __PYX_ERR(0, 46, __pyx_L4_error); __Pyx_GIVEREF(__pyx_t_4); if (__Pyx_PyTuple_SET_ITEM(__pyx_t_6, 1, __pyx_t_4)) __PYX_ERR(0, 46, __pyx_L4_error); __Pyx_GIVEREF(__pyx_t_5); if (__Pyx_PyTuple_SET_ITEM(__pyx_t_6, 2, __pyx_t_5)) __PYX_ERR(0, 46, __pyx_L4_error); __pyx_t_2 = 0; __pyx_t_4 = 0; __pyx_t_5 = 0; __pyx_t_5 = PyTuple_New(1); if (unlikely(!__pyx_t_5)) __PYX_ERR(0, 46, __pyx_L4_error) __Pyx_GOTREF(__pyx_t_5); __Pyx_GIVEREF(__pyx_t_6); if (__Pyx_PyTuple_SET_ITEM(__pyx_t_5, 0, __pyx_t_6)) __PYX_ERR(0, 46, __pyx_L4_error); __pyx_t_6 = 0; __pyx_t_6 = __Pyx_PyDict_NewPresized(1); if (unlikely(!__pyx_t_6)) __PYX_ERR(0, 46, __pyx_L4_error) __Pyx_GOTREF(__pyx_t_6); __Pyx_GetModuleGlobalName(__pyx_t_4, __pyx_n_s_np); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 46, __pyx_L4_error) __Pyx_GOTREF(__pyx_t_4); __pyx_t_2 = __Pyx_PyObject_GetAttrStr(__pyx_t_4, __pyx_n_s_float32); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 46, __pyx_L4_error) __Pyx_GOTREF(__pyx_t_2); __Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0; if (PyDict_SetItem(__pyx_t_6, __pyx_n_s_dtype, __pyx_t_2) < 0) __PYX_ERR(0, 46, __pyx_L4_error) __Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0; __pyx_t_2 = __Pyx_PyObject_Call(__pyx_t_3, __pyx_t_5, __pyx_t_6); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 46, __pyx_L4_error) __Pyx_GOTREF(__pyx_t_2); __Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0; __Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0; __Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0; __pyx_t_7 = __Pyx_PyObject_to_MemoryviewSlice_dsdsds_float(__pyx_t_2, PyBUF_WRITABLE); if (unlikely(!__pyx_t_7.memview)) __PYX_ERR(0, 46, __pyx_L4_error) __Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0; __pyx_v_blocks_stack = __pyx_t_7; __pyx_t_7.memview = NULL; __pyx_t_7.data = NULL;
+47: assert blocks_stack.shape[0] == n_rows * n_cols
#ifndef CYTHON_WITHOUT_ASSERTIONS
if (unlikely(__pyx_assertions_enabled())) {
__pyx_t_1 = ((__pyx_v_blocks_stack.shape[0]) == (__pyx_v_n_rows * __pyx_v_n_cols));
if (unlikely(!__pyx_t_1)) {
__Pyx_Raise(__pyx_builtin_AssertionError, 0, 0, 0);
__PYX_ERR(0, 47, __pyx_L4_error)
}
}
#else
if ((1)); else __PYX_ERR(0, 47, __pyx_L4_error)
#endif
+48: assert blocks_stack.shape[1] == block_rows_len
#ifndef CYTHON_WITHOUT_ASSERTIONS
if (unlikely(__pyx_assertions_enabled())) {
__pyx_t_1 = ((__pyx_v_blocks_stack.shape[1]) == __pyx_v_block_rows_len);
if (unlikely(!__pyx_t_1)) {
__Pyx_Raise(__pyx_builtin_AssertionError, 0, 0, 0);
__PYX_ERR(0, 48, __pyx_L4_error)
}
}
#else
if ((1)); else __PYX_ERR(0, 48, __pyx_L4_error)
#endif
+49: assert blocks_stack.shape[2] == block_cols_len
#ifndef CYTHON_WITHOUT_ASSERTIONS
if (unlikely(__pyx_assertions_enabled())) {
__pyx_t_1 = ((__pyx_v_blocks_stack.shape[2]) == __pyx_v_block_cols_len);
if (unlikely(!__pyx_t_1)) {
__Pyx_Raise(__pyx_builtin_AssertionError, 0, 0, 0);
__PYX_ERR(0, 49, __pyx_L4_error)
}
}
#else
if ((1)); else __PYX_ERR(0, 49, __pyx_L4_error)
#endif
+50: count = 0
__pyx_v_count = 0;
+51: for row_i in range(n_rows):
__pyx_t_8 = __pyx_v_n_rows;
__pyx_t_9 = __pyx_t_8;
for (__pyx_t_10 = 0; __pyx_t_10 < __pyx_t_9; __pyx_t_10+=1) {
__pyx_v_row_i = __pyx_t_10;
+52: for col_i in range(n_cols):
__pyx_t_11 = __pyx_v_n_cols;
__pyx_t_12 = __pyx_t_11;
for (__pyx_t_13 = 0; __pyx_t_13 < __pyx_t_12; __pyx_t_13+=1) {
__pyx_v_col_i = __pyx_t_13;
+53: blocks_stack[count] = img[row_i*block_rows_len:(row_i+1)*block_rows_len, col_i*block_cols_len:(col_i+1)*block_cols_len]
__pyx_t_14.data = __pyx_v_img.data;
__pyx_t_14.memview = __pyx_v_img.memview;
__PYX_INC_MEMVIEW(&__pyx_t_14, 1);
__pyx_t_15 = -1;
if (unlikely(__pyx_memoryview_slice_memviewslice(
&__pyx_t_14,
__pyx_v_img.shape[0], __pyx_v_img.strides[0], __pyx_v_img.suboffsets[0],
0,
0,
&__pyx_t_15,
(__pyx_v_row_i * __pyx_v_block_rows_len),
((__pyx_v_row_i + 1) * __pyx_v_block_rows_len),
0,
1,
1,
0,
1) < 0))
{
__PYX_ERR(0, 53, __pyx_L4_error)
}
if (unlikely(__pyx_memoryview_slice_memviewslice(
&__pyx_t_14,
__pyx_v_img.shape[1], __pyx_v_img.strides[1], __pyx_v_img.suboffsets[1],
1,
1,
&__pyx_t_15,
(__pyx_v_col_i * __pyx_v_block_cols_len),
((__pyx_v_col_i + 1) * __pyx_v_block_cols_len),
0,
1,
1,
0,
1) < 0))
{
__PYX_ERR(0, 53, __pyx_L4_error)
}
__pyx_t_16.data = __pyx_v_blocks_stack.data;
__pyx_t_16.memview = __pyx_v_blocks_stack.memview;
__PYX_INC_MEMVIEW(&__pyx_t_16, 1);
{
Py_ssize_t __pyx_tmp_idx = __pyx_v_count;
Py_ssize_t __pyx_tmp_stride = __pyx_v_blocks_stack.strides[0];
__pyx_t_16.data += __pyx_tmp_idx * __pyx_tmp_stride;
}
__pyx_t_16.shape[0] = __pyx_v_blocks_stack.shape[1];
__pyx_t_16.strides[0] = __pyx_v_blocks_stack.strides[1];
__pyx_t_16.suboffsets[0] = -1;
__pyx_t_16.shape[1] = __pyx_v_blocks_stack.shape[2];
__pyx_t_16.strides[1] = __pyx_v_blocks_stack.strides[2];
__pyx_t_16.suboffsets[1] = -1;
if (unlikely((__pyx_memoryview_copy_contents(__pyx_t_14, __pyx_t_16, 2, 2, 0) < 0))) __PYX_ERR(0, 53, __pyx_L4_error)
__PYX_XCLEAR_MEMVIEW(&__pyx_t_16, 1);
__pyx_t_16.memview = NULL; __pyx_t_16.data = NULL;
__PYX_XCLEAR_MEMVIEW(&__pyx_t_14, 1);
__pyx_t_14.memview = NULL; __pyx_t_14.data = NULL;
+54: count += 1
__pyx_v_count = (__pyx_v_count + 1);
}
}
}
55:
+56: return blocks_stack
__PYX_INC_MEMVIEW(&__pyx_v_blocks_stack, 0); __pyx_r = __pyx_v_blocks_stack; goto __pyx_L0;